import requests
import app


class EmployeeApi:

    def __init__(self):
        self.url_add_employee = app.base_url + '/api/sys/user'
        self.url_delete_employee = app.base_url + '/api/sys/user/{}'
        self.url_update_employee = app.base_url + ' /api/sys/user/{}'
        self.url_get_employee = app.base_url + ' /api/sys/user/{}'

    # 员工添加
    def add_employee(self, employee_data):
        return requests.post(url=self.url_add_employee, json=employee_data, headers=app.headers_data)

    # 员工修改
    def update_employee(self, employee_id, employee_data):
        url = self.url_update_employee.format(employee_id)
        return requests.put(url=url, json=employee_data, headers=app.headers_data)

    # 员工查询
    def get_employee(self, employee_id):
        url = self.url_get_employee.format(employee_id)
        return requests.get(url=url, headers=app.headers_data)

    # 员工删除
    def delete_employee(self, employee_id):
        url = self.url_delete_employee.format(employee_id)
        return requests.delete(url=url, headers=app.headers_data)
